package com.wt.admin.domain.entity.ai;


import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;
import com.wt.admin.config.JacksonTypeHandler;
import com.wt.admin.domain.entity.PublicEntity;
import lombok.Data;

import java.util.List;

@Data
@TableName(value = "ai_agents_info",autoResultMap = true)
public class AgentsInfoEntity extends PublicEntity {

    @TableId(value = "id",type = IdType.AUTO)
    private Integer id;

    @TableField(value = "avatar")
    private String avatar;

    @TableField(value = "name")
    private String name;

    @TableField(value = "background")
    private String background;

    @TableField(value = "config",typeHandler = JacksonTypeHandler.class)
    private Object config;


    @Data
    public static class Config{
        public Model model;
        public List<Integer> tools;
        public List<Integer> knowledges;
        public List<List<AgentSynergism>> synergism;
    }

    @Data
    public static class Model{
        public Integer chatModel;
        public Integer embeddingsModel;
    }

    @Data
    public static class AgentSynergism{
        public Integer agent;
        public String prompt;
    }

}
